from captum.attr import visualization as viz
from captum.attr import IntegratedGradients

ig = IntegratedGradients(None)
ig.attribute()